/*
 * To change this template, choose Tools | Templates
 * and open the template in the editor.
 */
package com.qp.smsmarketing.listagem;

import com.qp.smsmarketing.entity.Grupo;
import java.util.List;
import javax.swing.table.DefaultTableModel;

/**
 *
 * @author joasramos
 */
public class ListagemGrupo extends DefaultTableModel {

    private List<Grupo> listaGrupos = null;

    public ListagemGrupo(List<Grupo> lista) {
        this.listaGrupos = lista;
    }

    @Override
    public int getColumnCount() {
        return 3;
    }

    @Override
    public int getRowCount() {
        if (listaGrupos == null) {
            return 0;
        } else {
            return listaGrupos.size();
        }

    }

    @Override
    public Object getValueAt(int row, int column) {

        Grupo grupo = this.listaGrupos.get(row);
        Object retorno = null;

        if (column == 0) {
            retorno = grupo.getIdGrupo();
        } else if (column == 1) {
            retorno = grupo.getNomeGrupo();
        } else if (column == 2) {
            retorno = grupo.getGrupoContatoList().size();
        }
        return retorno;
    }

    @Override
    public String getColumnName(int column) {
        String nomeColuna = "";
        if (column == 0) {
            nomeColuna = "ID";
        } else if (column == 1) {
            nomeColuna = "Nome";
        } else if (column == 2) {
            nomeColuna = "Nº Contatos";
        }
        return nomeColuna;
    }
}
